from tqdm import tqdm
import time

total_steps = 3
total_iterations_per_step = 50

# 主循环，每个步骤都有一个嵌套的循环
for step in range(total_steps):
    # 使用tqdm创建进度条
    progress_bar = tqdm(total=total_iterations_per_step, desc=f"步骤 {step + 1}")

    # 内部循环，代表每个步骤中的计算
    for i in range(total_iterations_per_step):
        # 在这里执行你的计算步骤
        time.sleep(0.1)  # 模拟耗时操作

        # 更新进度条
        progress_bar.update(1)

    # 结束当前步骤的进度条
    progress_bar.close()

# 所有步骤完成
print("计算完成")